/*
 * Copyright (c) 2004-2022, University of Oslo
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * 3. Neither the name of the copyright holder nor the names of its contributors 
 * may be used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package org.hisp.dhis.dxf2.metadata.objectbundle.hooks;

import java.util.function.Consumer;
import org.hisp.dhis.common.Objects;
import org.hisp.dhis.common.ValueType;
import org.hisp.dhis.dxf2.metadata.objectbundle.ObjectBundle;
import org.hisp.dhis.feedback.ErrorCode;
import org.hisp.dhis.feedback.ErrorReport;
import org.hisp.dhis.textpattern.TextPattern;
import org.hisp.dhis.textpattern.TextPatternParser;
import org.hisp.dhis.textpattern.TextPatternValidationUtils;
import org.hisp.dhis.trackedentity.TrackedEntityAttribute;
import org.springframework.stereotype.Component;

@Component("org.hisp.dhis.dxf2.metadata.objectbundle.hooks.TrackedEntityAttributeObjectBundleHook")
public class TrackedEntityAttributeObjectBundleHook
    extends AbstractObjectBundleHook<TrackedEntityAttribute> {

  /**
   * Validate that the RenderType (if any) conforms to the constraints of ValueType or OptionSet.
   */
  @Override
  public void validate(
      TrackedEntityAttribute attr, ObjectBundle bundle, Consumer<ErrorReport> addReports) {
    if (attr.isGenerated() && !attr.getValueType().equals(ValueType.TEXT)) {
      addReports.accept(
          new ErrorReport(
              TrackedEntityAttribute.class, ErrorCode.E4010, "generated", attr.getValueType()));
    }

    textPatternValid(attr, addReports);

    if (attr.getFieldMask() != null) {
      try {
        TextPatternParser.parse("\"" + attr.getFieldMask() + "\"");
      } catch (TextPatternParser.TextPatternParsingException e) {
        addReports.accept(
            new ErrorReport(
                TrackedEntityAttribute.class,
                ErrorCode.E4019,
                attr.getFieldMask(),
                "Not a valid TextPattern 'TEXT' segment."));
      }
    }
  }

  @Override
  public void postCreate(TrackedEntityAttribute persistedObject, ObjectBundle bundle) {
    updateTextPattern(persistedObject);
  }

  @Override
  public void postUpdate(TrackedEntityAttribute persistedObject, ObjectBundle bundle) {
    updateTextPattern(persistedObject);
  }

  private void updateTextPattern(TrackedEntityAttribute attr) {
    if (attr.isGenerated()) {
      try {
        TextPattern textPattern = TextPatternParser.parse(attr.getPattern());
        textPattern.setOwnerObject(Objects.TRACKEDENTITYATTRIBUTE);
        textPattern.setOwnerUid(attr.getUid());
        attr.setTextPattern(textPattern);
      } catch (TextPatternParser.TextPatternParsingException e) {
        e.printStackTrace();
      }
    }
  }

  private void textPatternValid(TrackedEntityAttribute attr, Consumer<ErrorReport> addReports) {
    if (attr.isGenerated()) {
      try {
        TextPattern tp = TextPatternParser.parse(attr.getPattern());

        long generatedSegments =
            tp.getSegments().stream().filter((s) -> s.getMethod().isGenerated()).count();

        if (generatedSegments != 1) {
          addReports.accept(new ErrorReport(TrackedEntityAttribute.class, ErrorCode.E4021));
        }

        if (!TextPatternValidationUtils.validateValueType(tp, attr.getValueType())) {
          addReports.accept(
              new ErrorReport(
                  TrackedEntityAttribute.class,
                  ErrorCode.E4022,
                  attr.getPattern(),
                  attr.getValueType().name()));
        }
      } catch (TextPatternParser.TextPatternParsingException e) {
        addReports.accept(
            new ErrorReport(
                TrackedEntityAttribute.class, ErrorCode.E4019, attr.getPattern(), e.getMessage()));
      }
    }
  }
}
